Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op.cc76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute() local
77 OP_REQUIRES(context, kDims == dims, in Compute()
78 errors::InvalidArgument("Input rank should be: ", kDims, in Compute()
112 auto Tinput = input.tensor<T, kDims>(); in Compute()
113 auto Toutput = outputs_tensor->tensor<T, kDims>(); in Compute()
/external/skia/tests/
DProxyConversionTest.cpp130 static constexpr SkISize kDims = {64, 64}; in DEF_GPUTEST_FOR_RENDERING_CONTEXTS() local
136 format, kDims, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
150 format, kDims, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
164 format, kDims, GrRenderable::kNo, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
DGrSurfaceTest.cpp108 static constexpr SkISize kDims = {64, 64}; in DEF_GPUTEST_FOR_ALL_CONTEXTS() local
131 sk_sp<GrSurface> tex = createTexture(kDims, combo.fColorType, combo.fFormat, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
143 combo.fFormat, kDims, GrRenderable::kNo, 1, GrMipmapped::kYes, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
156 kDims, combo.fFormat, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBudgeted::kNo, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
169 kDims, combo.fFormat, GrRenderable::kYes, 2, GrMipmapped::kNo, SkBudgeted::kNo, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
DTextureBindingsResetTest.cpp77 static constexpr SkISize kDims = {10, 10}; in DEF_GPUTEST_FOR_GL_RENDERING_CONTEXTS() local
79 auto tex = gpu->createTexture(kDims, format, GrRenderable::kNo, 1, GrMipmapped::kNo, in DEF_GPUTEST_FOR_GL_RENDERING_CONTEXTS()
DOpChainTest.cpp178 static constexpr SkISize kDims = {kNumOps + 1, 1}; variable
185 format, kDims, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBackingFit::kExact,
DWritePixelsTest.cpp531 static constexpr SkISize kDims = {32, 64}; in DEF_GPUTEST_FOR_RENDERING_CONTEXTS() local
537 format, kDims, GrRenderable::kNo, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
DProcessorTest.cpp149 static constexpr SkISize kDims = {10, 10}; in DEF_GPUTEST_FOR_ALL_CONTEXTS() local
163 format, kDims, GrRenderable::kNo, 1, GrMipmapped::kNo, SkBackingFit::kExact, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc111 const int kDims = input_shape.dimensions_count(); in ExportFloatArray() local
113 kDims > 1 || (kDims == 1 && input_shape.dims(0) > 1)) { in ExportFloatArray()
114 for (int i = 0; i < kDims; ++i) { in ExportFloatArray()
2473 const int kDims = state_shape.dimensions_count(); in AddPlaceholderForRNNState() local
2474 for (int i = 0; i < kDims; ++i) { in AddPlaceholderForRNNState()