Home
last modified time | relevance | path

Searched defs:new_axis_mask (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc174 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
266 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
358 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc115 const int32 new_axis_mask = 0; in BM_ValidateStridedSliceOp() local
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc38 const int32 new_axis_mask, shrink_axis_mask; member
151 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp()
374 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc521 int32 new_axis_mask, in StridedSliceGrad()
562 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad()
616 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
643 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
671 int begin_mask = 4, end_mask = 2, new_axis_mask = 8, shrink_axis_mask = 1, in TEST() local
699 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 1, in TEST() local
Darray_ops.cc1453 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anon0d47aeef2602() local
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dstrided_slice_test.cc33 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/python/ops/
Dresource_variable_ops.py825 end_mask, ellipsis_mask, new_axis_mask, argument
Dvariables.py671 new_axis_mask, argument
Darray_ops.py654 new_axis_mask=0, argument
/external/tensorflow/tensorflow/contrib/lite/
Dbuiltin_op_data.h213 int new_axis_mask; member
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3095 int64 new_axis_mask = bitmask_distribution(generator()); in TEST_F() local
3143 int64 new_axis_mask = bitmask_distribution(generator()); in TEST_F() local