| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 57 TEST(TensorShapeTest, RemoveDim) { in TEST() argument
59 s.RemoveDim(0); in TEST()
66 s.RemoveDim(1); in TEST()
193 s2.RemoveDim(2); in TEST()
260 void RemoveDim(int d);
437 void TensorShapeOld::RemoveDim(int d) { in RemoveDim() function in tensorflow::__anone850e9110111::TensorShapeOld
552 s.RemoveDim(dim); in TEST()
553 sold.RemoveDim(dim); in TEST()
|
| D | lookup_interface.cc | 53 expected_value_shape.RemoveDim(expected_value_shape.dims() - 1); in CheckKeyAndValueTensorsHelper()
|
| D | tensor_shape.h | 206 void RemoveDim(int d) { in RemoveDim() function
|
| D | tensor.cc | 839 ret.shape_.RemoveDim(0); in SubSlice()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 345 element_shape_except_first_dim_.RemoveDim(0); in Compute()
519 tensor_shape_without_first_dim.RemoveDim(0); in Compute()
524 element_shape_without_first_dim.RemoveDim(0); in Compute()
690 output_shape.RemoveDim(0); in Compute()
700 tmp_shape.RemoveDim(0); in Compute()
723 tmp_shape.RemoveDim(0); in Scatter()
818 output_shape.RemoveDim(0); in Compute()
957 input_element_shape.RemoveDim(0); in Compute()
|
| D | unpack_op.cc | 65 output_shape.RemoveDim(axis); in Compute()
|
| D | queue_base.cc | 38 chip_shape.RemoveDim(0); in HandleSliceToElement()
|
| D | fifo_queue.cc | 93 element_shape.RemoveDim(0); in GetElementComponentFromBatch()
|
| D | priority_queue.cc | 121 element_shape.RemoveDim(0); in GetElementComponentFromBatch()
|
| D | padding_fifo_queue.cc | 301 chip_shape.RemoveDim(0); in ValidateElementToLargerSlice()
|
| D | random_shuffle_queue_op.cc | 169 element_shape.RemoveDim(0); in GetElementComponentFromBatch()
|
| D | barrier_ops.cc | 99 if (element_shape.dims() > 0) element_shape.RemoveDim(0); in TryInsertMany()
|
| D | tensor_array_ops.cc | 896 value_shape_t_except0.RemoveDim(0); in Compute()
1118 element_shape.RemoveDim(0); in Compute()
|
| D | encode_proto_op.cc | 540 shape_prefix.RemoveDim(shape_prefix.dims() - 1); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unbatch_dataset_op.cc | 140 shapes_[i].RemoveDim(0); in GetNextInternal()
204 shapes_[i].RemoveDim(0); in RestoreInternal()
|
| D | map_and_batch_dataset_op.cc | 456 batch_shape.RemoveDim(0); in CallFunction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unpack_op.cc | 63 output_shape.RemoveDim(axis); in Compile()
|
| D | tensor_list_ops.cc | 177 shape.RemoveDim(0); in Compile()
323 input_list_element_shape.RemoveDim(0); in GetInitializedList()
|
| D | tensor_array_ops.cc | 369 elem_shape.RemoveDim(0); in Compile()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | batch_util.cc | 34 chip_shape.RemoveDim(0); in ValidateInput()
197 chip_shape.RemoveDim(0); in ValidateElementToLargerSlice()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 281 gradients_shape.RemoveDim(0); in AddToTensorAccumulator()
282 hessians_shape.RemoveDim(0); in AddToTensorAccumulator()
776 gradients_shape.RemoveDim(0); in Compute()
781 hessians_shape.RemoveDim(0); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 292 arg_shapes.at(i).RemoveDim(0); in SetupArgs()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_seq_tensor.cc | 245 rest.RemoveDim(0); \
|