/external/tensorflow/tensorflow/core/framework/ |
partial_tensor_shape_test.cc | 153 EXPECT_TRUE(f.IsCompatibleWith(f)); 154 EXPECT_TRUE(a.IsCompatibleWith(b)); 155 EXPECT_TRUE(a.IsCompatibleWith(a)); 156 EXPECT_TRUE(b.IsCompatibleWith(b)); 157 EXPECT_TRUE(a.IsCompatibleWith(c)); 158 EXPECT_TRUE(b.IsCompatibleWith(c)); 159 EXPECT_FALSE(a.IsCompatibleWith(d)); 160 EXPECT_FALSE(b.IsCompatibleWith(d)); 161 EXPECT_FALSE(c.IsCompatibleWith(d)); 162 EXPECT_FALSE(a.IsCompatibleWith(e)) [all...] |
tensor_shape.h | 425 /// stronger condition that IsCompatibleWith. 430 bool IsCompatibleWith(const PartialTensorShape& shape) const;
|
tensor_shape.cc | 701 bool PartialTensorShape::IsCompatibleWith( 730 if (!shapes0[i].IsCompatibleWith(shapes1[i])) {
|
/external/tensorflow/tensorflow/core/kernels/ |
conditional_accumulator.h | 80 if (!shape_.IsCompatibleWith(tensor->shape())) {
|
padding_fifo_queue.cc | 241 if (!partial_shapes_[i].IsCompatibleWith(tuple[i].shape())) { 258 if (!expected_shape.IsCompatibleWith(tuple[i].shape())) {
|
list_kernels.h | 527 element_shape_without_first_dim.IsCompatibleWith( 691 OP_REQUIRES(c, element_shape.IsCompatibleWith(output_shape), [all...] |
shape_ops.cc | 474 if (!expected_shape_.IsCompatibleWith(shape)) {
|
tensor_array.h | 472 if (!element_shape_.IsCompatibleWith(value_t->shape())) {
|
list_kernels.cc | 282 OP_REQUIRES(c, l->element_shape.IsCompatibleWith(input.shape()), 490 OP_REQUIRES(c, l->element_shape.IsCompatibleWith(value.shape()),
|
example_parsing_ops.cc | 108 attrs_.dense_shapes[d].IsCompatibleWith(def_value.shape()), 209 attrs_.dense_shapes[d].IsCompatibleWith(def_value.shape()), [all...] |
tensor_array_ops.cc | 666 ctx, element_shape_.IsCompatibleWith(value_0_t->shape()), [all...] |
cudnn_rnn_ops.cc | 509 bool IsCompatibleWith(const CudnnRnnModelShapes& rhs) const { 548 return lhs.first.IsCompatibleWith(rhs.first) && lhs.second == rhs.second; [all...] |
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
parse_example_dataset_op.cc | 108 OP_REQUIRES(ctx, dense_shapes_[d].IsCompatibleWith(def_value.shape()), 301 DCHECK(dataset_->output_shapes()[output_index].IsCompatibleWith( 328 DCHECK(dataset_->output_shapes()[output_index].IsCompatibleWith(
|
directed_interleave_dataset_op.cc | 42 selector_input->output_shapes()[0].IsCompatibleWith(
|
scan_dataset_op.cc | 212 if (!output_shapes()[output_index].IsCompatibleWith(
|
indexed_dataset_op.cc | 381 ctx, expected_shapes[i].IsCompatibleWith(out_tensors[i].shape()),
|
/external/tensorflow/tensorflow/core/kernels/data/ |
optional_ops.cc | 125 OP_REQUIRES(ctx, output_shapes_[i].IsCompatibleWith(components[i].shape()),
|
dataset_utils.cc | 134 if (!expected[i].IsCompatibleWith(received[i])) {
|
map_defun_op.cc | 221 if (!compute_opts_->output_shapes.at(index).IsCompatibleWith(
|
iterator_ops.cc | 737 ctx, output_shapes_[i].IsCompatibleWith(state[i].shape()), [all...] |
/external/tensorflow/tensorflow/core/util/ |
example_proto_fast_parsing.cc | [all...] |
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
arithmetic_optimizer.cc | [all...] |
constant_folding.cc | [all...] |