Home
last modified time | relevance | path

Searched refs:size_as_vector (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass.cc73 std::vector<int64> size_as_vector; member
121 slice_inputs.size_as_vector = IntTensorAsVector(*tf_slice_size); in GetSliceInputs()
140 result.size_as_vector = slice_inputs.size_as_vector; in MakeSliceIndexAndSizeInt64()
184 if (absl::c_all_of(slice_inputs.size_as_vector, in ComputeSliceSize()
197 for (int i = 0, end = slice_inputs.size_as_vector.size(); i < end; i++) { in ComputeSliceSize()
198 if (slice_inputs.size_as_vector[i] >= 0) { in ComputeSliceSize()
200 constant_pool.Get1DHostConstant(slice_inputs.size_as_vector[i])); in ComputeSliceSize()
204 DCHECK_EQ(slice_inputs.size_as_vector[i], -1); in ComputeSliceSize()
337 slice_inputs->size_as_vector, [](int64 size_i) { return size_i >= -1; }); in ShouldRewriteSlice()