HomeSort by relevance Sort by last modified time
    Searched refs:stop_indices (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
resolve_strided_slice_attributes.cc 52 op->stop_indices = stop_array.GetBuffer<ArrayDataType::kInt32>().data;
57 CHECK_EQ(op->stop_indices.size(), op->start_indices.size());
58 CHECK_EQ(op->strides.size(), op->stop_indices.size());
resolve_constant_strided_slice.cc 51 stop = op.stop_indices[axis];
74 CHECK_EQ(num_input_axes, op.stop_indices.size());
149 if (op->start_indices.empty() || op->stop_indices.empty() ||
propagate_fixed_sizes.cc     [all...]
  /external/tensorflow/tensorflow/core/kernels/
strided_slice_op.h 35 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
43 stop_i[i] = stop_indices[i];
50 input.stridedSlice(start_indices, stop_indices, strides);
83 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
92 stop_i[i] = stop_indices[i];
98 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
109 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
117 stop_i[i] = stop_indices[i];
123 output.stridedSlice(start_indices, stop_indices, strides).device(d) =
  /external/tensorflow/tensorflow/contrib/lite/toco/tflite/
operator_test.cc 373 EXPECT_EQ(op.stop_indices, output_toco_op->stop_indices);
  /external/tensorflow/tensorflow/contrib/lite/toco/
model.h 804 std::vector<int> stop_indices; member in struct:toco::StridedSliceOperator
    [all...]
export_tensorflow.cc     [all...]

Completed in 221 milliseconds