Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h51 const dnn::BatchDescriptor& output_dimensions,
62 const dnn::BatchDescriptor& output_dimensions,
66 const dnn::BatchDescriptor& output_dimensions,
436 const dnn::BatchDescriptor& output_dimensions,
443 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
453 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
473 const dnn::BatchDescriptor& output_dimensions,
481 const dnn::BatchDescriptor& output_dimensions,
489 const dnn::BatchDescriptor& output_dimensions,
497 const dnn::BatchDescriptor& output_dimensions,
[all …]
Drocm_dnn.cc3815 const dnn::BatchDescriptor& output_dimensions, in DoMatMul() argument
3817 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
3835 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3836 output_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
3841 if (output_dimensions.width() == 1 && output_dimensions.height() == 1) { in DoMatMul()
3855 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3891 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3892 !(output_dimensions.layout() == dnn::DataLayout::kBatchDepthYX && in DoMatMul()
3893 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
3900 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h412 const dnn::BatchDescriptor& output_dimensions,
419 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
429 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
449 const dnn::BatchDescriptor& output_dimensions,
457 const dnn::BatchDescriptor& output_dimensions,
465 const dnn::BatchDescriptor& output_dimensions,
473 const dnn::BatchDescriptor& output_dimensions,
481 const dnn::BatchDescriptor& output_dimensions,
491 const dnn::BatchDescriptor& output_dimensions,
501 const dnn::BatchDescriptor& output_dimensions,
[all …]
Dcuda_dnn.cc4062 const dnn::BatchDescriptor& output_dimensions, in DoMatMul() argument
4064 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
4082 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
4083 output_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
4088 if (output_dimensions.width() == 1 && output_dimensions.height() == 1) { in DoMatMul()
4102 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
4138 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
4139 !(output_dimensions.layout() == dnn::DataLayout::kBatchDepthYX && in DoMatMul()
4140 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
4147 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc42 const int output_dimensions = NumElements(output_shape); in Resize() local
43 TfLiteIntArray* output_shape_array = TfLiteIntArrayCreate(output_dimensions); in Resize()
44 for (int i = 0; i < output_dimensions; ++i) { in Resize()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1567 const dnn::BatchDescriptor& output_dimensions,
1579 const dnn::BatchDescriptor& output_dimensions,
1591 const dnn::BatchDescriptor& output_dimensions,
1640 const dnn::BatchDescriptor& output_dimensions,
1648 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() argument
1659 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() argument
1670 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() argument
1682 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() argument
1695 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() argument
1708 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() argument
[all …]
Dstream.cc749 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul() argument
752 PARAM(output_dimensions), PARAM(output_data)); in ThenMatMul()
756 output_dimensions, output_data)); in ThenMatMul()
767 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized() argument
770 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
775 input_dimensions, output_dimensions, in ThenMatMulQuantized()
787 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized() argument
790 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
795 input_dimensions, output_dimensions, in ThenMatMulQuantized()
823 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward() argument
[all …]
Dstream.h518 const dnn::BatchDescriptor &output_dimensions,
525 const dnn::BatchDescriptor &output_dimensions,
532 const dnn::BatchDescriptor &output_dimensions,
543 const dnn::BatchDescriptor &output_dimensions,
550 const dnn::BatchDescriptor &output_dimensions,
557 const dnn::BatchDescriptor &output_dimensions,
564 const dnn::BatchDescriptor &output_dimensions,
571 const dnn::BatchDescriptor &output_dimensions,
580 const dnn::BatchDescriptor &output_dimensions,
589 const dnn::BatchDescriptor &output_dimensions,
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc171 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
209 output_dimensions[i] = window_util::StridedBound( in InferWindowOutputShape()
214 return ShapeUtil::MakeValidatedShape(element_type, output_dimensions, in InferWindowOutputShape()
750 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
754 output_dimensions[i] = lhs.dimensions(i); in InferDegenerateDimensionBroadcastShape()
756 output_dimensions[i] = rhs.dimensions(i); in InferDegenerateDimensionBroadcastShape()
758 output_dimensions[i] = lhs.dimensions(i); in InferDegenerateDimensionBroadcastShape()
775 output_dimensions, output_dimensions_is_dynamic); in InferDegenerateDimensionBroadcastShape()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc263 loc, op.getResult(), op.output_dimensions(), &rewriter); in matchAndRewrite()
319 b->create<tensor::ExtractOp>(loc, op.output_dimensions(), i_val); in InsertDynamicMemrefCastOp()
Dlegalize_to_linalg.cc500 Value shape = adaptor.output_dimensions(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6144 nvinfer1::Dims output_dimensions; in ConvertResize() local
6145 output_dimensions.nbDims = inputs_tensor->getDimensions().nbDims; in ConvertResize()
6146 for (int i = 0; i < output_dimensions.nbDims; ++i) { in ConvertResize()
6147 output_dimensions.d[i] = inputs_tensor->getDimensions().d[i]; in ConvertResize()
6149 output_dimensions.d[output_dimensions.nbDims - 2] = weights_ptr[0]; in ConvertResize()
6150 output_dimensions.d[output_dimensions.nbDims - 1] = weights_ptr[1]; in ConvertResize()
6160 layer->setOutputDimensions(output_dimensions); in ConvertResize()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td841 HLO_DimensionTensor:$output_dimensions,
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc843 op.output_dimensions().getType().cast<RankedTensorType>(); in Verify()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2813 `∀ i . 0 ≤ i < rank ⇒ input_dimensions[permutation[i]] = output_dimensions[i]`.