Home
last modified time | relevance | path

Searched defs:ellipsis_mask (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc182 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
Dstrided_slice_op_test.cc114 const int32 ellipsis_mask = 0; in BM_ValidateStridedSliceOp() local
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc37 int32 ellipsis_mask; member
155 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp()
378 int32 begin_mask_spec, int32 end_mask_spec, const int32 ellipsis_mask, in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc568 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
Darray_ops.cc1590 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, in __anon7c94107b2502() local
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc561 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
Dshape_refiner_test.cc67 int ellipsis_mask = 0) { in TestStridedSlice()
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice_test.cc34 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc415 int64 ellipsis_mask; in StridedSliceGradHelper() local
/external/tensorflow/tensorflow/python/ops/
Dresource_variable_ops.py1344 end_mask, ellipsis_mask, new_axis_mask, argument
Darray_ops.py733 ellipsis_mask=0, argument
Dvariables.py2121 ellipsis_mask, argument
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h309 int ellipsis_mask; member
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h949 int16 ellipsis_mask; member
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h994 int16 ellipsis_mask; member
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3222 int64 ellipsis_mask = 0; in TEST_F() local
3270 int64 ellipsis_mask = 0; in TEST_F() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1686 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3144 const int ellipsis_mask = node->attr().at("ellipsis_mask").i(); in GetSliceAxis() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2618 int64 new_axis_mask = 0, int64 shrink_axis_mask = 0) -> NodeDef { in TEST_F()