| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_gradient.py | 28 def get_gradients_through_compute_gradients(optimizer, loss, activations): argument
99 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 101 HloInstruction* activations = in TEST_F() local
143 HloInstruction* activations = in TEST_F() local
174 HloInstruction* activations = in TEST_F() local
204 HloInstruction* activations = in TEST_F() local
233 HloInstruction* activations = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 478 auto activations = convolution->mutable_operand(0); in IsThisBackPropFilterConv() local
510 HloInstruction* activations, in HaloDuplicateWithSlice()
640 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in BringSpaceNextToBatch()
744 HloInstruction* activations, int64_t batch_dimension, in SplitAndTransposeMergedBatch()
796 HloInstruction* activations, int64_t batch_dimension, in ChangeSpatialSizeOnSpaceToBatchedShape()
1173 auto activations = consumer->mutable_operand(0); in CanPropagate() local
2813 HloInstruction* activations, in TransposeAndMergeBatch()
2855 HloInstruction* activations, in PerformSplitSpace()
2901 HloInstruction* activations, in PadAndSplitSpace()
2936 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in SplitSpace()
[all …]
|
| D | dynamic_padder.cc | 1093 HloInstruction* activations = custom_call_conv->mutable_operand(0); in RewriteDynamicConvolutionKernelGrad() local
|
| D | memory_space_assignment_test.cc | 6387 auto activations = builder.AddInstruction(HloInstruction::CreateConstant( in TEST_P() local
6438 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/ops/ |
| D | nn_ops.cc | 88 AbstractTensorHandle** activations, const char* name, in Relu()
|
| /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 | 265 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 | 1231 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1257 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1284 auto activations = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1316 auto activations = in XLA_TEST_F() local
1358 auto activations = ConstantLiteral(&builder, activations_literal); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/python/tpu/tests/ |
| D | tpu_embedding_base_test.py | 630 def _get_total_loss_tensor(self, activations): argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 549 auto activations = orig_output_tensor.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 1592 std::vector<string> activations = {"Relu", "Relu6", "Elu", "Tanh", in RunTest() local
1595 std::vector<string> activations = {"Relu", "Relu6", "Elu", "LeakyRelu"}; in RunTest() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 54 namespace activations { namespace
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2356 std::vector<int64_t> activations = in TEST_F() local
2434 std::vector<int64_t> activations = in TEST_F() local
2556 std::vector<int64_t> activations = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1633 XlaOp activations, XlaOp gradients, in DynamicConvKernelGrad()
4341 XlaOp activations, XlaOp gradients, in DynamicConvKernelGrad()
|
| /external/crosvm/docs/book/ |
| D | mermaid.min.js | 24 …tion(t,e){t.exports=function(t){return t.webpackPolyfill||(t.deprecate=function(){},t.paths=[],t.c… argument
|