Searched refs:numDims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/go/ |
| D | attrs.go | 157 numDims := make([]C.int, meta.list_size)
160 …C.TF_OperationGetAttrShapeList(op.c, cname, &dims[0], &numDims[0], C.int(meta.list_size), &storage…
166 numDim := numDims[i]
228 numDims := meta.total_size
230 if numDims < 0 {
233 if numDims == 0 {
236 dims := make([]C.int64_t, numDims)
237 …C.TF_OperationGetAttrShape(op.c, cname, (*C.int64_t)(unsafe.Pointer(&dims[0])), C.int(numDims), st…
|
| D | tensor_handle.go | 76 n, err := th.numDims()
91 func (th *TensorHandle) numDims() (int, error) { func
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| D | OperationBuilder.java | 353 private static native void setAttrShape(long handle, String name, long[] shape, int numDims); in setAttrShape() argument
356 long handle, String name, long[] shapes, int[] numDims); in setAttrShapeList() argument
|
| /external/deqp-deps/glslang/hlsl/ |
| D | hlslParseHelper.cpp | 3943 int numDims = 0; in decomposeSampleMethods() local
3946 case Esd1D: numDims = 1; break; // W in decomposeSampleMethods()
3947 case Esd2D: numDims = 2; break; // W, H in decomposeSampleMethods()
3948 case Esd3D: numDims = 3; break; // W, H, D in decomposeSampleMethods()
3949 case EsdCube: numDims = 2; break; // W, H (cube) in decomposeSampleMethods()
3950 case EsdBuffer: numDims = 1; break; // W (buffers) in decomposeSampleMethods()
3951 case EsdRect: numDims = 2; break; // W, H (rect) in decomposeSampleMethods()
3958 ++numDims; in decomposeSampleMethods()
3962 const bool mipQuery = (numArgs > (numDims + 1 + (isMs ? 1 : 0))); in decomposeSampleMethods()
3985 sizeQuery->setType(TType(EbtUint, EvqTemporary, numDims)); in decomposeSampleMethods()
[all …]
|
| /external/deqp/modules/gles31/functional/ |
| D | es31fTextureGatherTests.cpp | 1243 const int numDims = getNumTextureSamplingDimensions(textureType); in genProgramSources() local
1246 …return glu::ProgramSources() << genVertexShaderSource(requireGpuShader5(gatherType), numDims, isDy… in genProgramSources()
1247 …<< genFragmentShaderSource(requireGpuShader5(gatherType), numDims, samplerType, funcCall, useNorma… in genProgramSources()
|
| /external/deqp/external/vulkancts/modules/vulkan/shaderrender/ |
| D | vktShaderRenderTextureGatherTests.cpp | 1642 const int numDims = getNumTextureSamplingDimensions(baseParams.textureType); in genGatherPrograms() local
1643 …rSource(requireGpuShader5(baseParams.gatherType, baseParams.offsetSize), numDims, isDynamicOffset … in genGatherPrograms()
1651 …rSource(requireGpuShader5(baseParams.gatherType, baseParams.offsetSize), numDims, samplerType, fun… in genGatherPrograms()
|