/external/tensorflow/tensorflow/core/kernels/ |
D | strided_slice_op.cc | 185 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp 279 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp 391 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
D | strided_slice_op_test.cc | 111 const int32 ellipsis_mask = 0; in BM_ValidateStridedSliceOp() local
|
/external/tensorflow/tensorflow/core/ops/ |
D | array_grad_test.cc | 568 int32 end_mask, int32 ellipsis_mask, in StridedSliceGrad() 610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() 665 ellipsis_mask = 0; in TEST() local 692 ellipsis_mask = 0; in TEST() local 720 ellipsis_mask = 0; in TEST() local 748 ellipsis_mask = 2; in TEST() local
|
D | array_ops.cc | 1729 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anondb9326b22502() local
|
/external/tensorflow/tensorflow/core/util/ |
D | strided_slice_op.cc | 39 int32 ellipsis_mask; member 172 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp() 420 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
D | lstm_utils.cc | 165 int64_t ellipsis_mask, int64_t new_axis_mask, in CreateStridedSliceOp() 778 int64_t ellipsis_mask = 0; in ConvertKerasLSTMLayer() local
|
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
D | types.h | 134 int16_t ellipsis_mask; member
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | shape_refiner.cc | 586 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
D | shape_refiner_test.cc | 67 int ellipsis_mask = 0) { in TestStridedSlice()
|
/external/tensorflow/tensorflow/cc/gradients/ |
D | array_grad.cc | 424 int64 ellipsis_mask; in StridedSliceGradHelper() local
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_ops_n_z.cc | 1628 uint32_t ellipsis_mask = op.ellipsis_mask(); in VerifyStridedSliceBase() local 1676 int32_t begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; member 1827 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in CalculateSlicedShapeFromSparseIndices()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | strided_slice_test.cc | 37 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | prepare_tf.cc | 597 uint64_t ellipsis_mask = strided_slice_op.ellipsis_mask(); in RewriteEllipsisMask() local
|
/external/tensorflow/tensorflow/lite/c/ |
D | builtin_op_data.h | 396 int ellipsis_mask; member
|
/external/tensorflow/tensorflow/python/ops/ |
D | resource_variable_ops.py | 1358 end_mask, ellipsis_mask, new_axis_mask, argument
|
D | array_ops.py | 1119 ellipsis_mask=0, argument
|
D | variables.py | 2542 end_mask, ellipsis_mask, new_axis_mask, argument
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 3223 int64 ellipsis_mask = 0; in TEST_F() local 3271 int64 ellipsis_mask = 0; in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
D | legalize_common.cc | 2002 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in convertStridedSliceOp()
|
/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | types.h | 1130 int16_t ellipsis_mask; member
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes_test.cc | 3602 int64 new_axis_mask = 0, int64 shrink_axis_mask = 0) -> NodeDef { in TEST_F() 3730 int ellipsis_mask; in TEST_F() member
|
D | convert_nodes.cc | 3119 const int32 ellipsis_mask = attrs.get<int64>("ellipsis_mask"); in ConvertStridedSlice() local
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | arithmetic_optimizer.cc | 3328 const int ellipsis_mask = node->attr().at("ellipsis_mask").i(); in GetStridedSliceAxis() local
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2517 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf.cc | 3200 int ellipsis_mask = op.ellipsis_mask(); in rewriteWithUnknownBegin() local
|