Searched refs:size_i (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 361 int64 size_i; in ValidateStridedSliceOp() local
365 size_i = 0; in ValidateStridedSliceOp()
367 size_i = interval_length / stride_i + in ValidateStridedSliceOp()
370 processing_shape->AddDim(size_i); in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 337 slice_inputs->size_as_vector, [](int64 size_i) { return size_i >= -1; }); in ShouldRewriteSlice() argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1147 ]).concatenate(_maybe_tensor_shape_from_tensor(size_i))),
1151 size_i) in enumerate(zip(flat_emit_dtypes, flat_emit_size))
1156 array_ops.zeros(_concat(batch_size, size_i), dtype_i)
1157 for size_i, dtype_i in zip(flat_emit_size, flat_emit_dtypes)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | barrier_ops_test.py | 140 size_i = b.incomplete_size()
160 self.assertEqual(self.evaluate(size_i),
174 self.assertEqual(self.evaluate(size_i),
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1569 for (auto size_i : llvm::enumerate(size)) { in Verify() local
1570 if (size_i.value().getSExtValue() < -1) { in Verify()
1582 int size_i = in Verify() local
1589 if (size_i >= 0 && begin_i + size_i > dim_i) { in Verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1799 int64_t size_i = 0; in CalculateSlicedShapeFromDenseIndices() local
1804 size_i = (interval_len / stride_i) + (interval_len % stride_i != 0); in CalculateSlicedShapeFromDenseIndices()
1813 input_shape[i] = size_i; in CalculateSlicedShapeFromDenseIndices()
|
| /external/llvm-project/polly/lib/External/ppcg/ |
| D | gpu.c | 762 isl_pw_aff *param, *size_i, *zero; in add_bounded_parameters_dynamic() local
768 size_i = isl_multi_pw_aff_get_pw_aff(size, i); in add_bounded_parameters_dynamic()
769 bound = isl_pw_aff_lt_set(isl_pw_aff_copy(param), size_i); in add_bounded_parameters_dynamic()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 4261 int64 size_i = (new_slice_limits[i] - new_slice_starts[i] + in HandleSlice() local
4264 new_slice_shape.set_dimensions(i, size_i); in HandleSlice()
|