| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_gradient.py | 32 def get_gradients_through_compute_gradients(optimizer, loss, activations): argument
103 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument
|
| D | tpu_embedding_v2_correctness_test.py | 618 def _get_total_loss_tensor(activations): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 100 HloInstruction* activations = in TEST_F() local
142 HloInstruction* activations = in TEST_F() local
173 HloInstruction* activations = in TEST_F() local
203 HloInstruction* activations = in TEST_F() local
232 HloInstruction* activations = in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | logistic.cc | 31 namespace activations { namespace
|
| D | tanh.cc | 32 namespace activations { namespace
|
| D | leaky_relu.cc | 28 namespace activations { namespace
|
| D | prelu.cc | 29 namespace activations { namespace
|
| D | elu.cc | 33 namespace activations { namespace
|
| D | activations.cc | 30 namespace activations { namespace
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 344 HloInstruction* activations, int64 spatial_dimension_to_split, in HaloDuplicateWithSlice()
479 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in BringSpaceNextToBatch()
570 HloInstruction* activations, int64 batch_dimension, int64 old_batch_size, in IncreaseSpatialSizeOnSpaceToBatchedShape()
800 auto activations = consumer->mutable_operand(0); in CanPropagate() local
2038 HloInstruction* activations, int64 spatial_dimension_to_split, in SplitSpaceHelper()
2091 ConvolutionVisitor::SplitSpace(HloInstruction* activations, in SplitSpace()
2593 auto activations = convolution->mutable_operand(0); in GetConvolutionDetails() local
2668 auto activations = convolution->mutable_operand(0); in PerformSpaceToBatchOnConvolution() local
|
| D | dynamic_padder.cc | 937 HloInstruction* activations = custom_call_conv->mutable_operand(0); in RewriteDynamicConvolutionKernelGrad() local
|
| D | memory_space_assignment_test.cc | 5041 auto activations = builder.AddInstruction(HloInstruction::CreateConstant( in TEST_P() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_functor.h | 111 typename TTypes<T>::Tensor activations; member
|
| D | lrn_op.cc | 405 auto activations = out_image.shaped<T, 2>({nodes * batch, depth}); in launch() local
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad.cc | 51 AbstractTensorHandle* activations = forward_outputs_[0]; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 261 auto activations = in Compile() local
|
| D | conv_op_helpers.cc | 392 StringPiece type_string, xla::XlaOp activations, in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_variants_test.cc | 1232 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1258 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1285 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1317 auto activations = in XLA_TEST_F() local
1363 auto activations = ConstantLiteral(&builder, activations_literal); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 641 std::vector<string> activations = {"Relu", "Relu6", "Elu", "Tanh", in RunTest() local
644 std::vector<string> activations = {"Relu", "Relu6", "Elu", "LeakyRelu"}; in RunTest() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 548 auto activations = orig_output_tensor.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 52 namespace activations { namespace
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1628 std::vector<int64> activations = in TEST_F() local
1702 std::vector<int64> activations = in TEST_F() local
1807 std::vector<int64> activations = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1552 XlaOp activations, XlaOp gradients, absl::Span<const int64> window_strides, in DynamicConvKernelGrad()
4387 XlaOp activations, XlaOp gradients, absl::Span<const int64> window_strides, in DynamicConvKernelGrad()
|