Searched defs:shapes (Results 1 - 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpack_op.cc47 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 Dbcast_ops.cc30 // 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 Dconcat_op.cc58 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 Dbcast_ops.cc24 // 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 Dsparse_concat_op.cc64 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 Dqueue_base.cc81 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 Drandom_dataset_op.cc69 static std::vector<PartialTensorShape>* shapes = variable
71 return *shapes;
H A Drange_dataset_op.cc63 static std::vector<PartialTensorShape>* shapes = variable
65 return *shapes;
H A Dreader_dataset_ops.cc104 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 DOperationBuilderTest.java153 // 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 Dpartial_tensor_shape_test.cc132 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 Dhvx_ops_support_checker_main.cc65 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 Dmatrix_band_part_op_test.py91 shapes = [ variable in class:MatrixBandPartBenchmark
109 for shape_ in self.shapes:
H A Ddeterminant_op_test.py163 shapes = [ variable in class:MatrixDeterminantBenchmark
187 for shape in self.shapes:
H A Dmatrix_exponential_op_test.py154 shapes = [ variable in class:MatrixExponentialBenchmark
178 for shape in self.shapes:
H A Dmatrix_inverse_op_test.py155 shapes = [ variable in class:MatrixInverseBenchmark
180 for shape in self.shapes:
H A Dmatrix_logarithm_op_test.py126 shapes = [ variable in class:MatrixLogarithmBenchmark
150 for shape in self.shapes:
H A Dcholesky_op_test.py197 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 Dgpu_layout_assignment_test.cc84 // 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 Dhlo_tfgraph_builder.cc155 // 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 Dkafka_dataset_ops.cc81 static std::vector<PartialTensorShape>* shapes = variable
83 return *shapes;
/external/gemmlowp/test/
H A Dbenchmark_meta_gemm.cc104 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 Dbenchmark_all_sizes.cc278 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 DOperationBuilder.java278 // 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 Doperation_builder_jni.cc267 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

Completed in 500 milliseconds

12