| /external/tensorflow/tensorflow/core/framework/ |
| D | partial_tensor_shape_test.cc | 153 EXPECT_TRUE(f.IsCompatibleWith(f)); in TEST()
154 EXPECT_TRUE(a.IsCompatibleWith(b)); in TEST()
155 EXPECT_TRUE(a.IsCompatibleWith(a)); in TEST()
156 EXPECT_TRUE(b.IsCompatibleWith(b)); in TEST()
157 EXPECT_TRUE(a.IsCompatibleWith(c)); in TEST()
158 EXPECT_TRUE(b.IsCompatibleWith(c)); in TEST()
159 EXPECT_FALSE(a.IsCompatibleWith(d)); in TEST()
160 EXPECT_FALSE(b.IsCompatibleWith(d)); in TEST()
161 EXPECT_FALSE(c.IsCompatibleWith(d)); in TEST()
162 EXPECT_FALSE(a.IsCompatibleWith(e)); in TEST()
[all …]
|
| D | tensor_shape.cc | 701 bool PartialTensorShape::IsCompatibleWith( in IsCompatibleWith() function in tensorflow::PartialTensorShape
730 if (!shapes0[i].IsCompatibleWith(shapes1[i])) { in AreCompatible()
|
| D | tensor_shape.h | 430 bool IsCompatibleWith(const PartialTensorShape& shape) const;
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 108 OP_REQUIRES(ctx, dense_shapes_[d].IsCompatibleWith(def_value.shape()), in MakeDataset()
301 DCHECK(dataset_->output_shapes()[output_index].IsCompatibleWith( in MapFunc()
328 DCHECK(dataset_->output_shapes()[output_index].IsCompatibleWith( in MapFunc()
|
| D | directed_interleave_dataset_op.cc | 42 selector_input->output_shapes()[0].IsCompatibleWith( in MakeDataset()
|
| D | scan_dataset_op.cc | 212 if (!output_shapes()[output_index].IsCompatibleWith( in GetNextInternal()
|
| D | indexed_dataset_op.cc | 381 ctx, expected_shapes[i].IsCompatibleWith(out_tensors[i].shape()), in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator.h | 80 if (!shape_.IsCompatibleWith(tensor->shape())) { in ValidateShape()
|
| D | padding_fifo_queue.cc | 241 if (!partial_shapes_[i].IsCompatibleWith(tuple[i].shape())) { in ValidateTuple()
258 if (!expected_shape.IsCompatibleWith(tuple[i].shape())) { in ValidateManyTuple()
|
| D | list_kernels.h | 527 element_shape_without_first_dim.IsCompatibleWith( in Compute()
691 OP_REQUIRES(c, element_shape.IsCompatibleWith(output_shape), in Compute()
819 OP_REQUIRES(c, element_shape.IsCompatibleWith(output_shape), in Compute()
868 if (!a.element_shape.IsCompatibleWith(b.element_shape)) { in TensorListBinaryAdd()
966 c, l->element_shape.IsCompatibleWith(input_element_shape), in Compute()
|
| D | list_kernels.cc | 282 OP_REQUIRES(c, l->element_shape.IsCompatibleWith(input.shape()), in Compute()
490 OP_REQUIRES(c, l->element_shape.IsCompatibleWith(value.shape()), in Compute()
|
| D | shape_ops.cc | 474 if (!expected_shape_.IsCompatibleWith(shape)) { in Compute()
|
| D | example_parsing_ops.cc | 108 attrs_.dense_shapes[d].IsCompatibleWith(def_value.shape()), in Compute()
209 attrs_.dense_shapes[d].IsCompatibleWith(def_value.shape()), in Compute()
|
| D | tensor_array.h | 472 if (!element_shape_.IsCompatibleWith(value_t->shape())) { in LockedWriteOrAggregate()
|
| D | tensor_array_ops.cc | 666 ctx, element_shape_.IsCompatibleWith(value_0_t->shape()), in Compute()
901 ctx, element_shape_except0_.IsCompatibleWith(output_shape_except0), in Compute()
|
| D | cudnn_rnn_ops.cc | 509 bool IsCompatibleWith(const CudnnRnnModelShapes& rhs) const { in IsCompatibleWith() function
548 return lhs.first.IsCompatibleWith(rhs.first) && lhs.second == rhs.second; in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 125 OP_REQUIRES(ctx, output_shapes_[i].IsCompatibleWith(components[i].shape()), in Compute()
|
| D | dataset_utils.cc | 134 if (!expected[i].IsCompatibleWith(received[i])) { in VerifyShapesCompatible()
|
| D | map_defun_op.cc | 221 if (!compute_opts_->output_shapes.at(index).IsCompatibleWith( in SetRetval()
|
| D | iterator_ops.cc | 737 ctx, output_shapes_[i].IsCompatibleWith(state[i].shape()), in ComputeAsync()
1031 ctx, output_shapes_[i].IsCompatibleWith(components[i].shape()), in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1793 if (!c.shape.IsCompatibleWith(c.default_value.shape())) { in FastParseSequenceExample()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 1552 return shape.IsCompatibleWith(new_dims); in IsSimplifiableReshape()
|
| D | arithmetic_optimizer.cc | 3272 if (input_slice_shape.IsCompatibleWith(output_shape)) { in RewriteGraph()
|