/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | pack_op.cc | 47 std::vector<TensorShape> shapes; variable 48 OP_REQUIRES_OK(ctx, ctx->InputList("values", &values, &shapes)); 54 // Verify that all input shapes match 56 OP_REQUIRES(ctx, shapes[0].IsSameSize(shapes[i]), 59 shapes[0].DebugString(), " != values[", i, "].shape = ", 60 shapes[i].DebugString())); 63 int expanded_num_dims = shapes[0].dims() + 1; 74 TensorShape child_shape(shapes[0]);
|
H A D | bcast_ops.cc | 30 // Given shapes of two tensors, computes the broadcast shape. 41 gtl::InlinedVector<BCast::Vec, 2> shapes; variable 49 shapes.push_back(BCast::Vec(shape.begin(), shape.end())); 51 BCast bcast(shapes[0], shapes[1]); 54 "Incompatible shapes: [", str_util::Join(shapes[0], ","), 55 "] vs. [", str_util::Join(shapes[1], ","), "]")); 73 // Given shapes of two tensors, computes the reduction indices for the 90 gtl::InlinedVector<BCast::Vec, 4> shapes; variable [all...] |
H A D | concat_op.cc | 58 std::vector<TensorShape> shapes; variable 59 OP_REQUIRES_OK(ctx, ctx->InputList("values", &values, &shapes)); 61 const int input_dims = shapes[0].dims(); 62 const TensorShape& input_shape = shapes[0]; 80 const TensorShape& in_shape = shapes[i]; 144 // which has the same number of dimensions. Their shapes match 148 // dimension, and their shapes are:
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | bcast_ops.cc | 24 // Given shapes of two tensors, computes the broadcast shape. 34 gtl::InlinedVector<BCast::Vec, 4> shapes; variable 44 shapes.push_back(vec); 46 BCast bcast(shapes[0], shapes[1]); 49 "Incompatible shapes: [", str_util::Join(shapes[0], ","), 50 "] vs. [", str_util::Join(shapes[1], ","), "]")); 69 // Given shapes of two tensors, computes the reduction indices for the 83 gtl::InlinedVector<BCast::Vec, 4> shapes; variable [all...] |
H A D | sparse_concat_op.cc | 64 OpInputList shapes; variable 65 OP_REQUIRES_OK(context, context->input_list("shapes", &shapes)); 66 OP_REQUIRES(context, shapes.size() == N, 67 errors::InvalidArgument("Expected ", N, " input shapes, got ", 68 shapes.size())); 70 OP_REQUIRES(context, TensorShapeUtils::IsVector(shapes[i].shape()), 72 "Input shapes should be a vector but received shape ", 73 shapes[i].shape().DebugString(), " at position ", i)); 76 const TensorShape input_shape(shapes[ [all...] |
H A D | queue_base.cc | 81 string QueueBase::ShapeListString(const gtl::ArraySlice<TensorShape>& shapes) { argument 84 for (const TensorShape& shape : shapes) { 131 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes)); 134 "' has component shapes ", 136 " but requested component shapes were ",
|
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | random_dataset_op.cc | 69 static std::vector<PartialTensorShape>* shapes = variable 71 return *shapes;
|
H A D | range_dataset_op.cc | 63 static std::vector<PartialTensorShape>* shapes = variable 65 return *shapes;
|
H A D | reader_dataset_ops.cc | 104 static std::vector<PartialTensorShape>* shapes = variable 106 return *shapes; 338 static std::vector<PartialTensorShape>* shapes = variable 340 return *shapes; 558 static std::vector<PartialTensorShape>* shapes = variable 560 return *shapes;
|
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
H A D | OperationBuilderTest.java | 153 // Those shapes match tensors ones, so no exception is thrown 156 // Those shapes do not match tensors ones, exception is thrown 191 private static void testSetAttrShapeList(Shape[] shapes) { argument 198 .setAttr("shapes", shapes)
|
/external/tensorflow/tensorflow/core/framework/ |
H A D | partial_tensor_shape_test.cc | 132 std::vector<PartialTensorShape> shapes = {a, b, c, d, e, f, g}; local 133 for (int i = 0; i < shapes.size(); ++i) { 136 EXPECT_TRUE(shapes[i].IsIdenticalTo(shapes[j])); 138 EXPECT_FALSE(shapes[i].IsIdenticalTo(shapes[j])); 248 // Empty made partial shapes should still be fully defined
|
/external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/ |
H A D | hvx_ops_support_checker_main.cc | 65 std::vector<TensorShape> shapes; local 67 node_def, &data_types, &shapes); 68 if (data_types.empty() || shapes.empty()) { 71 CHECK_EQ(data_types.size(), shapes.size()); 74 << ", " << shapes.at(i).DebugString();
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | matrix_band_part_op_test.py | 91 shapes = [ variable in class:MatrixBandPartBenchmark 109 for shape_ in self.shapes:
|
H A D | determinant_op_test.py | 163 shapes = [ variable in class:MatrixDeterminantBenchmark 187 for shape in self.shapes:
|
H A D | matrix_exponential_op_test.py | 154 shapes = [ variable in class:MatrixExponentialBenchmark 178 for shape in self.shapes:
|
H A D | matrix_inverse_op_test.py | 155 shapes = [ variable in class:MatrixInverseBenchmark 180 for shape in self.shapes:
|
H A D | matrix_logarithm_op_test.py | 126 shapes = [ variable in class:MatrixLogarithmBenchmark 150 for shape in self.shapes:
|
H A D | cholesky_op_test.py | 197 shapes = self.getShapes([1, 2, 10]) 199 shapes, dtypes=(dtypes_lib.float32, dtypes_lib.float64)) 203 shapes = self.getShapes([1, 2, 10]) 205 shapes, dtypes=(dtypes_lib.complex64, dtypes_lib.complex128)) 209 shapes = self.getShapes([self._backprop_block_size + 1]) 211 shapes, 217 shapes = self.getShapes([2 * self._backprop_block_size + 1]) 219 shapes, dtypes=(dtypes_lib.float32,), scalarTest=True) 223 shapes = self.getShapes([2 * self._backprop_block_size + 1]) 225 shapes, dtype 303 shapes = [ variable in class:CholeskyBenchmark [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_layout_assignment_test.cc | 84 // Returns a list shapes with all the possible layouts of this shape, including 90 std::vector<Shape> shapes; local 91 shapes.push_back(s); 92 shapes.back().clear_layout(); 95 shapes.push_back(s); 96 *shapes.back().mutable_layout() = LayoutUtil::MakeLayout(layout_vec); 99 return shapes; 186 // Enumerate all combinations of shapes. 256 // Enumerate all combinations of shapes plus whether we're constraining param
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_tfgraph_builder.cc | 155 // name used by Tensorboard for shapes of output tensors. 156 tensorflow::AttrValue shapes; local 157 *shapes.mutable_list()->add_shape() = GetTensorShape(instruction); 158 attrs["_output_shapes"] = shapes;
|
/external/tensorflow/tensorflow/contrib/kafka/kernels/ |
H A D | kafka_dataset_ops.cc | 81 static std::vector<PartialTensorShape>* shapes = variable 83 return *shapes;
|
/external/gemmlowp/test/ |
H A D | benchmark_meta_gemm.cc | 104 double run_gemms(std::vector<Shape>* shapes) { argument 106 for (auto& shape : *shapes) { 159 void time_all(std::vector<Shape>* shapes, std::int32_t repetitions, argument 169 ops += run_gemms(shapes);
|
H A D | benchmark_all_sizes.cc | 278 std::vector<Shape> shapes; local 286 shapes.push_back(shape); 288 // Benchmark an assortment of cubic shapes 294 shapes.push_back(shape); 297 // Benchmark all sorts of shapes 305 shapes.push_back(shape); 310 #error What shapes should we benchmark? (Suggestion: #define BENCHMARK_QUICK) 312 std::shuffle(std::begin(shapes), std::end(shapes), RandomEngine()); 313 return shapes; 317 std::vector<Shape> shapes; local [all...] |
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
H A D | OperationBuilder.java | 278 // Flatten the shapes into a single array to avoid too much overhead in the 280 long[] shapes = new long[totalNumDimensions]; 285 shapes[shapeIdx++] = dim; 291 setAttrShapeList(unsafeNativeHandle, name, shapes, numDimensions); 356 long handle, String name, long[] shapes, int[] numDims); 355 setAttrShapeList( long handle, String name, long[] shapes, int[] numDims) argument
|
/external/tensorflow/tensorflow/java/src/main/native/ |
H A D | operation_builder_jni.cc | 267 JNIEnv* env, jclass clazz, jlong handle, jstring name, jlongArray shapes, 276 const int shapes_length = env->GetArrayLength(shapes); 281 static_cast<jlong*>(env->GetPrimitiveArrayCritical(shapes, nullptr)); 283 env->ReleasePrimitiveArrayCritical(shapes, shapes_elems, JNI_ABORT); 266 Java_org_tensorflow_OperationBuilder_setAttrShapeList( JNIEnv* env, jclass clazz, jlong handle, jstring name, jlongArray shapes, jintArray num_dims) argument
|