| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batch_norm_op.cc | 124 auto activations = ctx->Input(1); variable
144 activations = b->ConvertElementType(activations, scale_type);
151 b->BatchNormGrad(activations, scale, mean, var, grad_backprop,
179 b->Mul(grad_backprop, b->Sub(activations, mean, {feature_index})),
|
| conv_ops.cc | 546 xla::ComputationDataHandle activations = ctx->Input(0); variable
550 // activations and the output gradients, with some appropriate padding.
555 // The activations (inputs) form the LHS of the convolution.
556 // Activations have shape: [batch, in_rows, in_cols, ..., in_depth]
561 // Swap n_dim and c_dim in the activations.
638 b->ConvGeneralDilated(activations, gradients, window_strides, padding,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_convolution_rewriter_test.cc | 50 // the batch and feature dimension in the activations, and treat the batch
97 HloInstruction* activations = local
99 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations"));
107 ShapeInference::InferConvolveShape(activations->shape(),
111 activations, gradients, conv_window,
126 HloInstruction* activations = local
128 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations"));
135 ShapeInference::InferConvolveShape(activations->shape(),
139 activations, gradients, conv_window,
155 HloInstruction* activations local
185 HloInstruction* activations = local
213 HloInstruction* activations = local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| convolution_variants_test.cc | 1234 auto activations = builder.ConstantR4FromArray4D<float>( local
1260 auto activations = builder.ConstantR4FromArray4D<float>( local
1287 auto activations = builder.ConstantR4FromArray4D<float>( local
1319 auto activations = local
1365 auto activations = builder.ConstantLiteral(*activations_literal); local
[all...] |
| vector_ops_simple_test.cc | 399 auto activations = builder.Map({x}, mult_relu_add, {0}); local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| activations.cc | 35 namespace activations { namespace in namespace:tflite::ops::builtin
340 } // namespace activations
344 activations::GenericPrepare,
345 activations::ReluEval};
351 activations::GenericPrepare,
352 activations::Relu1Eval};
358 activations::GenericPrepare,
359 activations::Relu6Eval};
365 activations::GenericPrepare,
366 activations::TanhEval}
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| lrn_op.cc | 323 auto activations = out_image.shaped<T, 2>({nodes * batch, depth}); local
328 auto shard = [this, activations, in_shaped, grads_shaped, out_shaped,
332 // Let y be the LRN activations and x be the inputs along the depth
359 activations(i, j) / norm;
|
| mkl_lrn_op.cc | 667 auto activations = out_image.shaped<T, 2>({nodes * batch, depth}); local
678 auto shard = [this, activations, in_shaped, grads_shaped, out_shaped,
693 activations(i, j) / norm;
[all...] |
| /prebuilts/tools/common/m2/repository/org/apache/maven/maven-model-builder/3.3.9/ |
| maven-model-builder-3.3.9.jar | |
| /prebuilts/tools/common/eclipse/ |
| org.eclipse.ui.workbench_3.6.2.M20110210-1200.jar | |
