Home
last modified time | relevance | path

Searched refs:Convolution (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter_test.cc58 EXPECT_THAT(reshape->operand(0)->operand(1), op::Convolution()); in TEST_F()
107 EXPECT_THAT(reshape->operand(0)->operand(1), op::Convolution()); in TEST_F()
141 EXPECT_THAT(reshape->operand(0)->operand(1), op::Convolution()); in TEST_F()
Dhlo_matchers.h222 HLO_MATCHER(Convolution);
Dhlo_cost_analysis_test.cc287 TEST_F(HloCostAnalysisTest, Convolution) { in TEST_F() argument
Dlayout_assignment_test.cc1041 GmockMatch(m::Convolution(m::Parameter(0), m::Parameter(1)))); in TEST_F()
Dpattern_matcher.h2081 XLA_BINOP_PATTERN(Convolution)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlayout_optimization_to_nhwc.mlir44 // Convolution layer #0.
87 // Convolution layer #1.
114 // Convolution layer #2.
/external/tensorflow/tensorflow/compiler/xla/tests/
Dsample_file_test.cc45 TEST_F(SampleFileTest, Convolution) { in TEST_F() argument
/external/XNNPACK/
DREADME.md17 - 2D Convolution (including grouped and depthwise)
18 - 2D Deconvolution (AKA Transposed Convolution)
92 - Marat Dukhan "The Indirect Convolution Algorithm". Presented on [Efficient Deep Learning for Comp…
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc632 AllOf(op::Convolution( in TEST_F()
1108 AllOf(op::Convolution( in TEST_F()
1157 AllOf(op::Convolution( in TEST_F()
1196 AllOf(op::Convolution( in TEST_F()
1236 EXPECT_THAT(root, AllOf(op::Convolution(sliced_lhs, rhs), in TEST_F()
1293 AllOf(op::DynamicSlice(AllOf(op::Convolution(masked, rhs), in TEST_F()
1527 EXPECT_THAT(root, AllOf(op::AllReduce(op::Convolution(lhs, rhs)), in TEST_F()
1556 AllOf(op::AllReduce(op::Convolution(lhs_masked, rhs_masked)), in TEST_F()
1590 EXPECT_THAT(root, AllOf(op::AllReduce(op::Convolution(reshard, rhs)), in TEST_F()
1626 AllOf(op::AllReduce(op::Convolution(op::Slice(lhs), reshard)), in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconv.cc38 class Convolution : public NodeShader { class
295 return absl::make_unique<Convolution>(); in NewConvolutionNodeShader()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
DREADME.md62 (asymmetric) versions of the following kernels: 1. Convolution 2D – Per axis
63 quantization only, `dilation_ratio==1` 2. Depthwise Convolution 2D – Per axis
/external/llvm/test/Transforms/InstCombine/
Dshufflemask-undef.ll13 …%struct.Convolution = type { %struct.IColor4, %struct.ImagingColorScale, i16, i16, [0 x i32], floa…
20 …%struct.ImagingSubset = type { %struct.Convolution, %struct.Convolution, %struct.Convolution, %str…
/external/llvm-project/llvm/test/Transforms/InstCombine/X86/
Dshufflemask-undef.ll14 …%struct.Convolution = type { %struct.IColor4, %struct.ImagingColorScale, i16, i16, [0 x i32], floa…
21 …%struct.ImagingSubset = type { %struct.Convolution, %struct.Convolution, %struct.Convolution, %str…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops_structs.td35 let summary = "GPU Convolution backend configuration";
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto103 // Convolution-specific parameters.
/external/python/cpython3/Modules/_decimal/libmpdec/literature/
Dbignum.txt18 Convolution in pseudo-code:
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/
Dspace_to_depth.md113 transform, input of Convolution needs to be padded as well.
129 tf\_device::LaunchOp and get the first Convolution and its shape;
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py224 convolver1 = nn_ops.Convolution(
230 convolver2 = nn_ops.Convolution(
Dconv_ops_test.py467 convolver1 = nn_ops.Convolution(
473 convolver2 = nn_ops.Convolution(
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/
DTosaOps.td83 let summary = "2D Convolution Operator";
114 let summary = "3D Convolution operator";
144 let summary = "Depthwise 2D Convolution operator";
252 let summary = "Transpose 2D Convolution operator.";
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/g3doc/
Dconv_emitter.md1 # Convolution Emitter
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto175 // Disable GEMM and Convolution auto-tuning.
Dxla_data.proto433 // The actual padding value depends upon the context. Convolution pads with
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc118 class Convolution class
1855 ops.push_back(MakeUnique<Convolution>(::tflite::BuiltinOperator_CONV_2D, in BuildOperatorList()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1158 op = Convolution(
1170 class Convolution(object): class

12