| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_grad_test.cc | 569 int32_t new_axis_mask, in StridedSliceGrad()
611 int32_t new_axis_mask, int32_t shrink_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
|
| H A D | array_ops.cc | 1762 int32_t begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anon38bbb0e82502() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op.cc | 188 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
282 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
409 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| H A D | strided_slice_op_test.cc | 112 const int32_t new_axis_mask = 0; in BM_ValidateStridedSliceOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 41 const int32 new_axis_mask, shrink_axis_mask; member
174 int32_t new_axis_mask, int32_t shrink_axis_mask, in ValidateStridedSliceOp()
426 int32_t new_axis_mask, int32_t shrink_axis_mask, in ValidateStridedSliceOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | slice_spmd_expander.cc | 175 const uint64_t new_axis_mask = strided_slice.new_axis_mask(); in GetInputOrientedData() local
391 const llvm::SmallVector<int64_t, 4> new_axis_mask = in ApplyNewAndShrinkMasksToLayout() local
619 const llvm::SmallVector<int64_t, 4> new_axis_mask = in ExpandOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | lstm_utils.cc | 168 int64_t ellipsis_mask, int64_t new_axis_mask, in CreateStridedSliceOp()
785 int64_t new_axis_mask = 0; in ConvertKerasLSTMLayer() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | prepare_tf.cc | 425 uint64_t new_axis_mask = strided_slice_op.new_axis_mask(); in RewriteNewAxisMask() local
507 uint64_t new_axis_mask = strided_slice_op.new_axis_mask(); in RewriteEllipsisMask() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | eval_const_tensor.cc | 160 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in TryToInferTensorOutputFromStridedSliceNode() local
|
| H A D | shape_refiner.cc | 607 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | strided_slice_test.cc | 38 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad.cc | 439 int64_t new_axis_mask; in StridedSliceGradHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 1816 int32_t begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; member
1968 int32_t new_axis_mask, int32_t shrink_axis_mask, in CalculateSlicedShapeFromSparseIndices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | builtin_op_data.h | 400 int new_axis_mask; member
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | builtin_op_data.h | 408 int new_axis_mask; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | resource_variable_ops.py | 1431 end_mask, ellipsis_mask, new_axis_mask, argument
|
| H A D | array_ops.py | 1176 new_axis_mask=0, argument
|
| H A D | variables.py | 2536 end_mask, ellipsis_mask, new_axis_mask, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 2586 std::bitset<32> new_axis_mask(0); in ConvertSlice() local
2660 int32 begin_mask, end_mask, ellipsis_mask, shrink_axis_mask, new_axis_mask; in ConvertStridedSlice() local
|
| H A D | convert_nodes_test.cc | 5073 int64 shrink_axis_mask = 0) -> NodeDef { in TEST_P()
5123 int new_axis_mask; in TEST_P() member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | types.h | 980 uint16_t new_axis_mask; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 4515 int64_t new_axis_mask = bitmask_distribution(generator()); in TEST_F() local
4565 int64_t new_axis_mask = bitmask_distribution(generator()); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_common.cc | 2193 int32_t new_axis_mask, int32_t shrink_axis_mask) { in convertStridedSliceOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 3204 uint32_t new_axis_mask = op.new_axis_mask(); in verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer.cc | 3913 const int new_axis_mask = node->attr().at("new_axis_mask").i(); in GetStridedSliceAxis() local
|