| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_v2_op_test.cc | 136 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
152 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
168 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
184 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
200 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
216 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
233 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
249 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
265 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
281 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
[all …]
|
| D | save_op_test.cc | 145 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
164 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
183 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
202 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
221 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
239 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
257 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
275 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
293 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
311 EXPECT_TRUE(shape.IsSameSize(expected)); in TEST_F()
[all …]
|
| D | restore_v2_op_test.cc | 211 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
223 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
235 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
247 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
259 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
271 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
284 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
296 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
308 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
320 EXPECT_TRUE(output->shape().IsSameSize(expected)); in RunTest()
[all …]
|
| D | restore_op_test.cc | 201 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
213 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
225 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
237 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
249 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
261 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
274 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
286 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
298 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
310 EXPECT_TRUE(output->shape().IsSameSize(expected)); in TEST_F()
[all …]
|
| D | training_ops.cc | 957 ctx, var.shape().IsSameSize(delta.shape()), in Compute()
1083 ctx, var.shape().IsSameSize(accum.shape()), in DoValidate()
1088 ctx, var.shape().IsSameSize(grad.shape()), in DoValidate()
1208 ctx, var.shape().IsSameSize(accum_grad.shape()), in DoCompute()
1212 OP_REQUIRES(ctx, var.shape().IsSameSize(accum_update.shape()), in DoCompute()
1353 ctx, var.shape().IsSameSize(delta.shape()), in Compute()
1564 ctx, var.shape().IsSameSize(accum.shape()), in Compute()
1569 ctx, var.shape().IsSameSize(grad.shape()), in Compute()
1669 ctx, var.shape().IsSameSize(accum.shape()), in Compute()
1674 ctx, var.shape().IsSameSize(grad.shape()), in Compute()
[all …]
|
| D | xent_op.cc | 48 if (!logits_in.IsSameSize(labels_in)) { in Compute()
79 if (logits_in.IsSameSize(labels_in)) { in Compute()
|
| D | save_restore_tensor.cc | 110 OP_REQUIRES(context, slice_shape.IsSameSize(input.shape()), in SaveTensors()
203 context, parsed_shape.IsSameSize(saved_shape), in RestoreTensor()
292 if (!restored_full_shape.IsSameSize(parsed_full_shape)) { in run()
|
| D | fake_quant_ops.cc | 135 OP_REQUIRES(context, input.IsSameSize(gradient), in OperateNoTemplate()
244 OP_REQUIRES(context, input.IsSameSize(gradient), in Compute()
383 OP_REQUIRES(context, input.IsSameSize(gradient), in Compute()
|
| D | queue_base.cc | 148 if (!component_shapes_[i].IsSameSize(tuple[i].shape())) { in ValidateTuple()
168 if (!expected_shape.IsSameSize(tuple[i].shape())) { in ValidateManyTuple()
|
| D | assign_op.h | 66 const bool same_shape = old_lhs.shape().IsSameSize(rhs.shape()); in Compute()
|
| D | conditional_accumulator.h | 73 if (!accum_grad_->shape().IsSameSize(tensor->shape())) { in ValidateShape()
|
| D | softplus_op.cc | 72 OP_REQUIRES(context, a.IsSameSize(g), in OperateNoTemplate()
|
| D | softsign_op.cc | 73 OP_REQUIRES(context, a.IsSameSize(g), in OperateNoTemplate()
|
| D | dense_update_ops.cc | 77 context, Tparams.IsSameSize(Tupdate), in DoUpdate()
|
| D | initializable_lookup_table.h | 180 if (!key_shape.IsSameSize(values_->shape())) { in KeyValueTensorIterator()
|
| D | pack_op.cc | 100 OP_REQUIRES(c, first_input.shape().IsSameSize(input.shape()), in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 46 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()
247 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), in Compile()
258 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
296 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), in Compile()
312 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
[all …]
|
| D | select_op.cc | 41 ctx, then_shape.IsSameSize(else_shape), in Compile()
50 bool broadcasting = !cond_shape.IsSameSize(then_shape); in Compile()
|
| D | pack_op.cc | 56 OP_REQUIRES(ctx, shapes[0].IsSameSize(shapes[i]), in Compile()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | summary_op.cc | 65 bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2);
77 if (!IsSameSize(params.tags, params.values)) { in ScalarSummaryOp_Compute()
113 bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2) { in IsSameSize() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 208 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({5, 7}))); in TEST()
219 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({5, 7}))); in TEST()
230 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({10, 20}))); in TEST()
329 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
340 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
351 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
362 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 2}))); in TEST()
381 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({2, 3, 4, 5}))); in SetUp()
382 EXPECT_TRUE(zero_t.shape().IsSameSize(TensorShape({3, 0, 2, 0, 5}))); in SetUp()
908 EXPECT_TRUE(t.shape().IsSameSize(TensorShape({10, 4, 3, 2}))); in TEST()
[all …]
|
| D | tensor_shape.h | 367 bool IsSameSize(const TensorShape& b) const;
368 bool operator==(const TensorShape& b) const { return IsSameSize(b); }
369 bool operator!=(const TensorShape& b) const { return !IsSameSize(b); }
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | ops.cc | 80 if (!TensorShape{e.tensor.shape()}.IsSameSize( in Initializer()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 101 if (!lhs_tf_shape.IsSameSize(rhs_tf_shape)) { in BroadcastOpsToSame()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_set.cc | 125 if (!shape.IsSameSize(tss_shape)) { in RegisterTensorSlice()
|