| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 182 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
275 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
373 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| D | strided_slice_op_test.cc | 115 const int32 new_axis_mask = 0; in BM_ValidateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 38 const int32 new_axis_mask, shrink_axis_mask; member
156 int32 new_axis_mask, int32 shrink_axis_mask, in ValidateStridedSliceOp()
379 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 569 int32 new_axis_mask, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
719 int begin_mask = 4, end_mask = 2, new_axis_mask = 8, shrink_axis_mask = 1, in TEST() local
747 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 1, in TEST() local
|
| D | array_ops.cc | 1590 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anon7c94107b2502() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 34 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 416 int64 new_axis_mask; in StridedSliceGradHelper() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 1344 end_mask, ellipsis_mask, new_axis_mask, argument
|
| D | array_ops.py | 734 new_axis_mask=0, argument
|
| D | variables.py | 2122 new_axis_mask, argument
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 310 int new_axis_mask; member
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 951 int16 new_axis_mask; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 996 int16 new_axis_mask; member
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3229 int64 new_axis_mask = bitmask_distribution(generator()); in TEST_F() local
3277 int64 new_axis_mask = bitmask_distribution(generator()); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1686 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3145 const int new_axis_mask = node->attr().at("new_axis_mask").i(); in GetSliceAxis() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2618 int64 new_axis_mask = 0, int64 shrink_axis_mask = 0) -> NodeDef { in TEST_F()
|