Searched refs:RemoveDim (Results 1 - 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dunpack_op.cc63 output_shape.RemoveDim(axis);
H A Dtensor_array_ops.cc370 elem_shape.RemoveDim(0);
/external/tensorflow/tensorflow/core/kernels/
H A Dbitcast_op.cc59 adjusted_shape.RemoveDim(input_tensor.dims() - 1);
H A Dbatch_util.cc34 chip_shape.RemoveDim(0);
140 chip_shape.RemoveDim(0);
H A Dunpack_op.cc65 output_shape.RemoveDim(axis);
H A Dlist_kernels.h149 output_shape.RemoveDim(0);
159 tmp_shape.RemoveDim(0);
H A Dfifo_queue.cc93 element_shape.RemoveDim(0);
H A Dqueue_base.cc38 chip_shape.RemoveDim(0);
H A Dpadding_fifo_queue.cc301 chip_shape.RemoveDim(0);
H A Dpriority_queue.cc121 element_shape.RemoveDim(0);
H A Dbarrier_ops.cc99 if (element_shape.dims() > 0) element_shape.RemoveDim(0);
H A Drandom_shuffle_queue_op.cc169 element_shape.RemoveDim(0);
H A Dtensor_array_ops.cc862 value_shape_t_except0.RemoveDim(0);
1084 element_shape.RemoveDim(0);
/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape_test.cc57 TEST(TensorShapeTest, RemoveDim) {
59 s.RemoveDim(0);
66 s.RemoveDim(1);
193 s2.RemoveDim(2);
253 void RemoveDim(int d);
430 void TensorShapeOld::RemoveDim(int d) { function in class:tensorflow::__anon26276::TensorShapeOld
545 s.RemoveDim(dim);
546 sold.RemoveDim(dim);
H A Dlookup_interface.cc53 expected_value_shape.RemoveDim(expected_value_shape.dims() - 1);
H A Dtensor_shape.h206 void RemoveDim(int d) { function in class:tensorflow::TensorShapeBase
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dstats_accumulator_ops.cc281 gradients_shape.RemoveDim(0);
282 hessians_shape.RemoveDim(0);
774 gradients_shape.RemoveDim(0);
779 hessians_shape.RemoveDim(0);
/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_and_batch_dataset_op.cc301 batch_shape.RemoveDim(0);
H A Dtensor_queue_dataset_op.cc622 element_shape.RemoveDim(0);
/external/tensorflow/tensorflow/python/lib/core/
H A Dpy_seq_tensor.cc174 rest.RemoveDim(0); \

Completed in 527 milliseconds