Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_test.cc45 int kDim = 100; in SliceHelper() local
54 end.flat<int32>()(0) = 10 + kDim; in SliceHelper()
61 Tensor input(dt, TensorShape({2 * kDim, kMaxSize})); in SliceHelper()
74 state.SetBytesProcessed(static_cast<int64>(state.iterations()) * kDim * size * in SliceHelper()
97 int kDim = 100; in BM_ValidateStridedSliceOp() local
101 Tensor end = test::AsTensor<int32>({10 + kDim, 10 + size}); in BM_ValidateStridedSliceOp()
103 TensorShape input_shape({2 * kDim, kMaxSize}); in BM_ValidateStridedSliceOp()
Dslice_op_test.cc44 int kDim = 100; in SliceHelper() local
53 sizes.flat<int32>()(0) = kDim; in SliceHelper()
56 Tensor input(dt, TensorShape({2 * kDim, kMaxSize})); in SliceHelper()
71 state.SetBytesProcessed(static_cast<int64>(state.iterations()) * kDim * size * in SliceHelper()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dunpack_quantize_test.cc49 const int kDim = 2; in PrepareModel() local
62 for (int i = 0; i < kDim; i++) { in PrepareModel()
88 out_array_shape_dim->resize(kDim - 1); in PrepareModel()
89 for (int i = 0; i < kDim - 1; i++) { in PrepareModel()
Dresolve_constant_concatenation_test.cc115 const int kDim = 3; in PrepareModel() local
132 for (int i = 0; i < kDim; i++) { in PrepareModel()
151 out_array_shape_dim->resize(kDim); in PrepareModel()
152 for (int i = 0; i < kDim; i++) { in PrepareModel()
/external/perfetto/src/base/
Dlogging.cc37 const char kDim[] = "\x1b[2m"; variable
94 color = kDim; in LogMessage()
/external/perfetto/src/tracing/
Dconsole_interceptor.cc80 const char kDim[] = "\x1b[90m"; variable
184 track_color.g, track_color.b, kReset, kDim, title_width, in OnTrackUpdated()
204 SetColor(context_, kDim); in OnTrackEvent()
246 SetColor(context_, kDim); in OnTrackEvent()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc516 const int32 kDim = operand_shape[2]; in RunR3Contiguous() local
517 Array3D<T> input_values(kSeq, kBatch, kDim); in RunR3Contiguous()
518 Array3D<T> update_values(size, kBatch, kDim); in RunR3Contiguous()
519 Array3D<T> expected_values(kSeq, kBatch, kDim); in RunR3Contiguous()
531 for (int k = 0; k < kDim; k++) { in RunR3Contiguous()