Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
Dperiodic_resample_op.h36 InputIndexer(const std::vector<tensorflow::int64>& output_dimensions, in InputIndexer() argument
39 : output_dimensions_(output_dimensions), in InputIndexer()
49 ComputeDimensionCeiling(output_dimensions, input_dimensions); in InputIndexer()
81 const std::vector<tensorflow::int64>& output_dimensions,
181 const std::vector<tensorflow::int64>& output_dimensions, in ComputeDimensionCeiling() argument
185 dimension_ceiling[i] = (output_dimensions[i] + input_dimensions[i] - 1) / in ComputeDimensionCeiling()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h393 const dnn::BatchDescriptor& output_dimensions,
400 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
410 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
430 const dnn::BatchDescriptor& output_dimensions,
438 const dnn::BatchDescriptor& output_dimensions,
446 const dnn::BatchDescriptor& output_dimensions,
454 const dnn::BatchDescriptor& output_dimensions,
462 const dnn::BatchDescriptor& output_dimensions,
472 const dnn::BatchDescriptor& output_dimensions,
482 const dnn::BatchDescriptor& output_dimensions,
[all …]
Dcuda_dnn.cc3529 const dnn::BatchDescriptor& output_dimensions, in DoMatMul() argument
3531 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
3549 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3550 output_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
3555 if (output_dimensions.width() == 1 && output_dimensions.height() == 1) { in DoMatMul()
3569 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3605 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3606 !(output_dimensions.layout() == dnn::DataLayout::kBatchDepthYX && in DoMatMul()
3607 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
3614 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h384 const dnn::BatchDescriptor& output_dimensions,
391 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
401 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() argument
421 const dnn::BatchDescriptor& output_dimensions,
429 const dnn::BatchDescriptor& output_dimensions,
437 const dnn::BatchDescriptor& output_dimensions,
445 const dnn::BatchDescriptor& output_dimensions,
455 const dnn::BatchDescriptor& output_dimensions,
465 const dnn::BatchDescriptor& output_dimensions,
495 const dnn::BatchDescriptor& output_dimensions,
Drocm_dnn.cc3289 const dnn::BatchDescriptor& output_dimensions, in DoMatMul() argument
3291 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
3309 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3310 output_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
3315 if (output_dimensions.width() == 1 && output_dimensions.height() == 1) { in DoMatMul()
3329 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3365 if (output_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3366 !(output_dimensions.layout() == dnn::DataLayout::kBatchDepthYX && in DoMatMul()
3367 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
3374 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc46 const int output_dimensions = NumElements(output_shape); in Resize() local
47 TfLiteIntArray* output_shape_array = TfLiteIntArrayCreate(output_dimensions); in Resize()
48 for (int i = 0; i < output_dimensions; ++i) { in Resize()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1501 const dnn::BatchDescriptor& output_dimensions,
1513 const dnn::BatchDescriptor& output_dimensions,
1525 const dnn::BatchDescriptor& output_dimensions,
1574 const dnn::BatchDescriptor& output_dimensions,
1582 const dnn::BatchDescriptor& output_dimensions,
1593 const dnn::BatchDescriptor& output_dimensions,
1604 const dnn::BatchDescriptor& output_dimensions,
1616 const dnn::BatchDescriptor& output_dimensions,
1629 const dnn::BatchDescriptor& output_dimensions,
1642 const dnn::BatchDescriptor& output_dimensions,
[all …]
Dstream.cc1142 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul() argument
1145 PARAM(output_dimensions), PARAM(output_data)); in ThenMatMul()
1150 output_dimensions, output_data)); in ThenMatMul()
1162 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized() argument
1165 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
1172 output_dimensions, output_data)); in ThenMatMulQuantized()
1184 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized() argument
1187 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
1194 output_dimensions, output_data)); in ThenMatMulQuantized()
1224 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward() argument
[all …]
Dstream.h495 const dnn::BatchDescriptor &output_dimensions,
502 const dnn::BatchDescriptor &output_dimensions,
509 const dnn::BatchDescriptor &output_dimensions,
520 const dnn::BatchDescriptor &output_dimensions,
527 const dnn::BatchDescriptor &output_dimensions,
534 const dnn::BatchDescriptor &output_dimensions,
541 const dnn::BatchDescriptor &output_dimensions,
548 const dnn::BatchDescriptor &output_dimensions,
557 const dnn::BatchDescriptor &output_dimensions,
566 const dnn::BatchDescriptor &output_dimensions,
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc169 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
214 output_dimensions[i] = window_util::StridedBound( in InferWindowOutputShape()
219 return ShapeUtil::MakeValidatedShape(element_type, output_dimensions, in InferWindowOutputShape()
720 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
724 output_dimensions[i] = lhs.dimensions(i); in InferDegenerateDimensionBroadcastShape()
727 output_dimensions[i] = rhs.dimensions(i); in InferDegenerateDimensionBroadcastShape()
730 output_dimensions[i] = lhs.dimensions(i); in InferDegenerateDimensionBroadcastShape()
740 output_dimensions, output_dimensions_is_dynamic); in InferDegenerateDimensionBroadcastShape()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2507 `∀ i . 0 ≤ i < rank ⇒ input_dimensions[permutation[i]] = output_dimensions[i]`.