Home
last modified time | relevance | path

Searched refs:IsSameSize (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dsave_v2_op_test.cc134 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
150 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
166 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
182 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
198 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
214 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
231 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
247 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
263 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
279 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
[all …]
Dsave_op_test.cc142 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
161 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
180 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
199 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
218 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
236 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
254 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
272 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
290 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
308 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
[all …]
Drestore_v2_op_test.cc208 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
220 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
232 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
244 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
256 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
268 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
281 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
293 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
305 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
317 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
[all …]
Drestore_op_test.cc198 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
210 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
222 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
234 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
246 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
258 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
271 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
283 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
295 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
307 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
[all …]
Dtraining_ops.cc485 ctx, var.shape().IsSameSize(delta.shape()), in Compute()
527 ctx, var.shape().IsSameSize(delta.shape()), in Compute()
672 ctx, var.shape().IsSameSize(accum.shape()), in DoValidate()
677 ctx, var.shape().IsSameSize(grad.shape()), in DoValidate()
801 ctx, var.shape().IsSameSize(accum_grad.shape()), in DoCompute()
805 OP_REQUIRES(ctx, var.shape().IsSameSize(accum_update.shape()), in DoCompute()
948 ctx, var.shape().IsSameSize(delta.shape()), in Compute()
1159 ctx, var.shape().IsSameSize(accum.shape()), in Compute()
1164 ctx, var.shape().IsSameSize(grad.shape()), in Compute()
1248 ctx, var.shape().IsSameSize(accum.shape()), in Compute()
[all …]
Dxent_op.cc51 if (!logits_in.IsSameSize(labels_in)) { in Compute()
82 if (logits_in.IsSameSize(labels_in)) { in Compute()
Dsave_restore_tensor.cc110 OP_REQUIRES(context, slice_shape.IsSameSize(input.shape()), in SaveTensors()
202 context, parsed_shape.IsSameSize(saved_shape), in RestoreTensor()
291 if (!restored_full_shape.IsSameSize(parsed_full_shape)) { in run()
Dfake_quant_ops.cc125 OP_REQUIRES(context, input.IsSameSize(gradient), in OperateNoTemplate()
232 OP_REQUIRES(context, input.IsSameSize(gradient), in Compute()
369 OP_REQUIRES(context, input.IsSameSize(gradient), in Compute()
Dqueue_base.cc148 if (!component_shapes_[i].IsSameSize(tuple[i].shape())) { in ValidateTuple()
168 if (!expected_shape.IsSameSize(tuple[i].shape())) { in ValidateManyTuple()
Dassign_op.h66 const bool same_shape = old_lhs.shape().IsSameSize(rhs.shape()); in Compute()
Dconditional_accumulator.h73 if (!accum_grad_->shape().IsSameSize(tensor->shape())) { in ValidateShape()
Dcwise_op_select.cc100 ctx, then->shape().IsSameSize(else_->shape()), in ComputeBroadcasting()
133 ctx, then->shape().IsSameSize(else_->shape()), in ComputeScalar()
Dsoftsign_op.cc73 OP_REQUIRES(context, a.IsSameSize(g), in OperateNoTemplate()
Dsoftplus_op.cc72 OP_REQUIRES(context, a.IsSameSize(g), in OperateNoTemplate()
Dinitializable_lookup_table.h172 if (!key_shape.IsSameSize(values_->shape())) { in KeyValueTensorIterator()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc46 ctx, var_shape.IsSameSize(delta_shape), in Compile()
98 ctx, var_shape.IsSameSize(delta_shape), in Compile()
131 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), in Compile()
142 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
189 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), in Compile()
200 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
245 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), in Compile()
256 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
287 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), in Compile()
305 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
[all …]
Dselect_op.cc39 ctx, then_shape.IsSameSize(else_shape), in Compile()
48 bool broadcasting = !cond_shape.IsSameSize(then_shape); in Compile()
Dpack_op.cc56 OP_REQUIRES(ctx, shapes[0].IsSameSize(shapes[i]), in Compile()
/external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/
Dinception_preprocessing_test.cc77 EXPECT_TRUE(outputs[0].shape().IsSameSize({1, 224, 224, 3})); in TEST()
106 EXPECT_TRUE(outputs[0].shape().IsSameSize({1, 224, 224, 3})); in TEST()
135 EXPECT_TRUE(outputs[0].shape().IsSameSize({1, 224, 224, 3})); in TEST()
Dimagenet_topk_eval.cc53 if (!output.shape().IsSameSize({1, kNumCategories})) { in ComputeEval()
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc185 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({5, 7}))); in TEST()
196 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({5, 7}))); in TEST()
207 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({10, 20}))); in TEST()
299 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
310 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
321 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
332 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
351 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 3, 4, 5}))); in SetUp()
352 EXPECT_TRUE(zero_t.shape().IsSameSize(TensorShape({3, 0, 2, 0, 5}))); in SetUp()
875 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({10, 4, 3, 2}))); in TEST()
[all …]
Dtensor_shape.h299 bool IsSameSize(const TensorShape& b) const;
300 bool operator==(const TensorShape& b) const { return IsSameSize(b); }
301 bool operator!=(const TensorShape& b) const { return !IsSameSize(b); }
Dtensor_shape_test.cc286 bool IsSameSize(const TensorShapeOld& b) const;
287 bool operator==(const TensorShapeOld& b) const { return IsSameSize(b); } in operator ==()
456 bool TensorShapeOld::IsSameSize(const TensorShapeOld& b) const { in IsSameSize() function in tensorflow::__anone850e9110111::TensorShapeOld
/external/tensorflow/tensorflow/lite/tools/accuracy/
Dutils_test.cc50 EXPECT_TRUE(shape.IsSameSize({1, 8, 8, 3})); in TEST()
/external/tensorflow/tensorflow/cc/framework/
Dops.cc80 if (!TensorShape{e.tensor.shape()}.IsSameSize( in Initializer()

12