Home
last modified time | relevance | path

Searched refs:RemoveDim (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc58 TEST(TensorShapeTest, RemoveDim) { in TEST() argument
60 s.RemoveDim(0); in TEST()
67 s.RemoveDim(1); in TEST()
194 s2.RemoveDim(2); in TEST()
283 void RemoveDim(int d);
460 void TensorShapeOld::RemoveDim(int d) { in RemoveDim() function in tensorflow::__anon127dd62f0111::TensorShapeOld
575 s.RemoveDim(dim); in TEST()
576 sold.RemoveDim(dim); in TEST()
Dlookup_interface.cc53 expected_value_shape.RemoveDim(expected_value_shape.dims() - 1); in CheckKeyAndValueTensorsHelper()
88 fullsize_value_shape.RemoveDim(fullsize_value_shape.dims() - 1); in CheckFindArguments()
Dtensor_shape.h241 void RemoveDim(int d) { in RemoveDim() function
Dtensor.cc893 ret.shape_.RemoveDim(0); in SubSlice()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h321 element_shape_except_first_dim.RemoveDim(0); in Compute()
496 tensor_shape_without_first_dim.RemoveDim(0); in Compute()
501 element_shape_without_first_dim.RemoveDim(0); in Compute()
667 output_shape.RemoveDim(0); in Compute()
677 tmp_shape.RemoveDim(0); in Compute()
700 tmp_shape.RemoveDim(0); in Scatter()
795 output_shape.RemoveDim(0); in Compute()
946 input_element_shape.RemoveDim(0); in Compute()
Dunpack_op.cc62 output_shape.RemoveDim(axis); in Compute()
Dqueue_base.cc38 chip_shape.RemoveDim(0); in HandleSliceToElement()
Dfifo_queue.cc93 element_shape.RemoveDim(0); in GetElementComponentFromBatch()
Dpadding_fifo_queue.cc301 chip_shape.RemoveDim(0); in ValidateElementToLargerSlice()
Dpriority_queue.cc121 element_shape.RemoveDim(0); in GetElementComponentFromBatch()
Drandom_shuffle_queue_op.cc170 element_shape.RemoveDim(0); in GetElementComponentFromBatch()
Dbarrier_ops.cc99 if (element_shape.dims() > 0) element_shape.RemoveDim(0); in TryInsertMany()
Dtensor_array_ops.cc890 value_shape_t_except0.RemoveDim(0); in Compute()
1111 element_shape.RemoveDim(0); in Compute()
Dencode_proto_op.cc560 shape_prefix.RemoveDim(shape_prefix.dims() - 1); in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dunpack_op.cc63 output_shape.RemoveDim(axis); in Compile()
Dtensor_array_ops.cc369 elem_shape.RemoveDim(0); in Compile()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc164 shapes_[i].RemoveDim(0); in GetNextInternal()
229 shapes_[i].RemoveDim(0); in RestoreInternal()
Drebatch_dataset_op.cc498 remaining_shape.RemoveDim(0); in GetNextInternal()
Dmap_and_batch_dataset_op.cc441 batch_shape.RemoveDim(0); in CallFunction()
/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc34 chip_shape.RemoveDim(0); in ValidateInput()
319 chip_shape.RemoveDim(0); in ValidateElementToLargerSlice()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc282 arg_shapes.at(i).RemoveDim(0); in SetupArgs()