Home
last modified time | relevance | path

Searched refs:end_mask (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1763 …,begin_mask=None,constant_indices=False,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1764 …,begin_mask=None,constant_indices=False,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1765 …0],begin_mask=None,constant_indices=False,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
1766 …0],begin_mask=None,constant_indices=False,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
1767 …0],begin_mask=None,constant_indices=False,dtype=tf.int64,end=[12,2,2,5],end_mask=None,index_type=t…
1768 …0],begin_mask=None,constant_indices=False,dtype=tf.int64,end=[12,2,2,5],end_mask=None,index_type=t…
1769 …],begin_mask=None,constant_indices=True,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1770 …],begin_mask=None,constant_indices=True,dtype=tf.float32,end=[12,2,2,5],end_mask=None,index_type=t…
1771 …,0],begin_mask=None,constant_indices=True,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
1772 …,0],begin_mask=None,constant_indices=True,dtype=tf.int32,end=[12,2,2,5],end_mask=None,index_type=t…
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h62 p->end_mask <<= pad_count; in StridedSlicePadIndices()
64 p->end_mask |= (1 << pad_count) - 1; in StridedSlicePadIndices()
115 const auto end_mask = params.end_mask; in StopForAxis() local
133 if (end_mask & (1 << axis)) { in StopForAxis()
170 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
187 op_params.end_mask = end_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc36 const int32 begin_mask, end_mask; member
49 int32 end_mask; member
81 dense->end_mask = 0; in BuildDenseSpec()
109 dense->end_mask |= (1 << full_index); in BuildDenseSpec()
133 if (sparse.end_mask & (1 << i)) { in BuildDenseSpec()
134 dense->end_mask |= (1 << full_index); in BuildDenseSpec()
271 {dense_spec.begin_mask & (1 << i), dense_spec.end_mask & (1 << i)}}; in ValidateStridedSliceOp()
292 (dense_spec.begin_mask & (1 << i)) && (dense_spec.end_mask & (1 << i)); in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc568 int32 end_mask, int32 ellipsis_mask, in StridedSliceGrad() argument
585 {"end_mask", end_mask}, in StridedSliceGrad()
610 int32 begin_mask, int32 end_mask, int32 ellipsis_mask, int32 new_axis_mask, in StridedSliceGradGrad() argument
628 {"end_mask", end_mask}, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
667 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
679 begin_mask, end_mask, ellipsis_mask, in TEST()
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
694 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
706 begin_mask, end_mask, ellipsis_mask, in TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc84 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceOp()
103 context->input(0).shape(), begin_mask, end_mask, in Compute()
181 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
191 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceGradOp()
229 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
274 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
284 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceAssignOp()
325 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
372 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
Dstrided_slice_op_test.cc113 const int32 end_mask = 0; in BM_ValidateStridedSliceOp() local
119 &begin, &end, strides, input_shape, begin_mask, end_mask, ellipsis_mask, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc93 return op_context->params->end_mask & (1 << idx) in GetEndValueAtIndex()
203 int end_mask = op_context.params->end_mask << (4 - op_context.dims); in Eval() local
208 begin_mask, end_mask, shrink_axis_mask, starts, stops, strides); in Eval()
Dstrided_slice_test.cc34 int end_mask, int ellipsis_mask, int new_axis_mask, in StridedSliceOpModel() argument
43 CreateStridedSliceOptions(builder_, begin_mask, end_mask, ellipsis_mask, in StridedSliceOpModel()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt17 `end[i]` is like `begin` with the exception that `end_mask` is
40 name: "end_mask"
104 `-1` if `stride < 0`. `end_mask` is analogous but produces the number
127 end_mask = 1<<5 = 32
161 `end_mask` are also set.
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc113 op->end_mask = in Run()
114 PadAttributeArray(&stop_array, input_array.shape().dims(), op->end_mask); in Run()
Dresolve_constant_strided_slice.cc57 op.begin_mask, op.end_mask, op.shrink_axis_mask, op.start_indices, in StridedSlice()
/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
564 TF_RETURN_IF_ERROR(GetNodeAttr(slice_node->attrs(), "end_mask", &end_mask)); in PartialStridedSliceShape()
575 !(end_mask == 0 || end_mask == 1) || ellipsis_mask != 0 || in PartialStridedSliceShape()
595 if (end_mask == 1) { in PartialStridedSliceShape()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1064 int end_mask;
1079 end_mask = other.end_mask;
1095 end_mask |= 1 << i;
1112 end_mask = toco::port::ReverseBits32(static_cast<uint32>(end_mask)) >>
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc2596 key: "end_mask" in CreateImagenetDatasetFunctions()
2721 key: "end_mask" in CreateImagenetDatasetFunctions()
2903 key: "end_mask" in CreateImagenetDatasetFunctions()
3048 key: "end_mask" in CreateImagenetDatasetFunctions()
3282 key: "end_mask" in CreateImagenetDatasetFunctions()
3427 key: "end_mask" in CreateImagenetDatasetFunctions()
3720 key: "end_mask" in CreateImagenetDatasetFunctions()
3862 key: "end_mask" in CreateImagenetDatasetFunctions()
4004 key: "end_mask" in CreateImagenetDatasetFunctions()
4541 key: "end_mask" in CreateImagenetDatasetFunctions()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py598 begin_mask, end_mask = 0, 0
616 end_mask |= (1 << index)
664 end_mask=end_mask,
732 end_mask=0, argument
832 end_mask=end_mask,
856 end_mask=end_mask,
Dresource_variable_ops.py1344 end_mask, ellipsis_mask, new_axis_mask, argument
1356 end_mask=end_mask,
/external/v8/src/heap/
Dslot-set.h114 uint32_t end_mask = ~((1u << end_bit) - 1); in RemoveRange() local
119 ClearCellBits(&bucket[start_cell], ~(start_mask | end_mask)); in RemoveRange()
167 ClearCellBits(&bucket[end_cell], ~end_mask); in RemoveRange()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc458 op.end_mask = 2; in TEST_F()
469 EXPECT_EQ(op.end_mask, output_toco_op->end_mask); in TEST_F()
470 EXPECT_EQ(op.end_mask, output_toco_op->end_mask); in TEST_F()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc414 int64 end_mask; in StridedSliceGradHelper() local
420 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "end_mask", &end_mask)); in StridedSliceGradHelper()
430 .EndMask(end_mask) in StridedSliceGradHelper()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h308 int end_mask; member
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py910 end_mask = tensor.op.get_attr("end_mask")
911 if end_mask == 1:
920 (not end_mask or (end_mask == 1)))
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1850 end_mask = pfor_input.get_attr("end_mask")
1859 end_mask = end_mask << 1 | 1
1870 end_mask=end_mask,
1884 end_mask = pfor_input.get_attr("end_mask")
1894 end_mask = end_mask << 1 | 1
1906 end_mask=end_mask,
/external/pdfium/third_party/libpng16/
Dpngrutil.c3065 unsigned int end_mask; in png_combine_row() local
3091 end_mask = (pixel_depth * row_width) & 7; in png_combine_row()
3092 if (end_mask != 0) in png_combine_row()
3100 end_mask = 0xff << end_mask; in png_combine_row()
3104 end_mask = 0xff >> end_mask; in png_combine_row()
3536 *end_ptr = (png_byte)((end_byte & end_mask) | (*end_ptr & ~end_mask)); in png_combine_row()
/external/libpng/
Dpngrutil.c3188 unsigned int end_mask; in png_combine_row() local
3214 end_mask = (pixel_depth * row_width) & 7; in png_combine_row()
3215 if (end_mask != 0) in png_combine_row()
3223 end_mask = (unsigned int)(0xff << end_mask); in png_combine_row()
3227 end_mask = 0xff >> end_mask; in png_combine_row()
3659 *end_ptr = (png_byte)((end_byte & end_mask) | (*end_ptr & ~end_mask)); in png_combine_row()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc542 params->end_mask = schema_params->end_mask(); in ParseOpData()

123