Searched refs:temp_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | mpi_ops.cc | 428 Tensor temp_tensor; in PerformCollectiveOp() local
449 temp_tensor = record.temp_t; in PerformCollectiveOp()
491 context, input_tensor, &temp_tensor, output_tensor) in PerformCollectiveOp()
493 context, input_tensor, &temp_tensor, output_tensor); in PerformCollectiveOp()
496 context, input_tensor, &temp_tensor, output_tensor) in PerformCollectiveOp()
498 context, input_tensor, &temp_tensor, output_tensor); in PerformCollectiveOp()
501 context, input_tensor, &temp_tensor, output_tensor) in PerformCollectiveOp()
503 context, input_tensor, &temp_tensor, output_tensor); in PerformCollectiveOp()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops.cc | 429 Tensor temp_tensor; in PerformCollectiveOp() local
450 temp_tensor = record.temp_t; in PerformCollectiveOp()
492 context, input_tensor, &temp_tensor, output_tensor) in PerformCollectiveOp()
494 context, input_tensor, &temp_tensor, output_tensor); in PerformCollectiveOp()
497 context, input_tensor, &temp_tensor, output_tensor) in PerformCollectiveOp()
499 context, input_tensor, &temp_tensor, output_tensor); in PerformCollectiveOp()
502 context, input_tensor, &temp_tensor, output_tensor) in PerformCollectiveOp()
504 context, input_tensor, &temp_tensor, output_tensor); in PerformCollectiveOp()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_custom.md | 155 TfLiteTensor* temp_tensor = &context->tensors[op_data->temp_tensor_index];
156 temp_tensor->type = kTfLiteFloat32;
157 temp_tensor->allocation_type = kTfLiteArenaRw;
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 886 TensorProto temp_tensor; in GetTensorShapeProtoFromTensorProto() local
887 tensor.AsProtoField(&temp_tensor); in GetTensorShapeProtoFromTensorProto()
891 for (const auto& value : temp_tensor.type##_val()) { \ in GetTensorShapeProtoFromTensorProto()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3819 const nvinfer1::ITensor* temp_tensor = nullptr; in ConvertMatMulHelper() local
3824 &temp_tensor)); in ConvertMatMulHelper()
3825 output_tensor = const_cast<nvinfer1::ITensor*>(temp_tensor); in ConvertMatMulHelper()
|