HomeSort by relevance Sort by last modified time
    Searched refs:set_dtype (Results 1 - 25 of 41) sorted by null

1 2

  /external/tensorflow/tensorflow/core/kernels/
unique_op_test.cc 40 tensor_proto.set_dtype(DT_INT32);
53 tensor_proto.set_dtype(DT_INT32);
107 tensor_proto.set_dtype(DT_STRING);
remote_fused_graph_execute_op_test.cc 149 shape_a.set_dtype(DT_FLOAT);
156 shape_a_plus_b.set_dtype(DT_FLOAT);
  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator_test.cc 34 input->set_dtype(DT_FLOAT);
78 input->set_dtype(dtype);
107 input->set_dtype(DT_FLOAT);
136 output->set_dtype(DT_FLOAT);
analytical_cost_estimator.cc 103 output_info->set_dtype(output.dtype());
utils.cc 57 input.set_dtype(DataType::DT_INVALID);
192 input.set_dtype(output.dtype());
  /external/tensorflow/tensorflow/core/profiler/internal/
tfprof_tensor.cc 36 tfprof_tensor_pb_.set_dtype(tensor_->dtype());
  /external/tensorflow/tensorflow/core/grappler/clusters/
virtual_cluster.cc 86 output_info->set_dtype(output.dtype());
  /external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transfer_utils.cc 72 tensor_shape_type.set_dtype(input.second.dtype());
87 tensor_shape_type_proto.set_dtype(tensor_shape_type->first);
hexagon_graph_execution_test.cc 217 tensor_shape_type.set_dtype(input.dtype());
229 tensor_shape_type.set_dtype(output.dtype());
graph_transferer.cc 148 graph_input_node_info.set_dtype(input_node_info.second.dtype());
173 graph_output_node_info.set_dtype(data_types.at(port));
491 const_node_info.set_dtype(const_tensor.dtype());
544 const_node_info.set_dtype(tensor.dtype());
    [all...]
  /external/tensorflow/tensorflow/core/framework/
tensor.cc 599 Tensor::Tensor(DataType type) : shape_({0}), buf_(nullptr) { set_dtype(type); }
603 set_dtype(type);
634 set_dtype(other_dtype);
715 set_dtype(type);
729 set_dtype(type);
788 ret.set_dtype(dtype());
828 set_dtype(proto.dtype());
843 proto->set_dtype(dtype());
851 proto->set_dtype(dtype());
    [all...]
variant_op_copy_test.cc 175 variant_proto.set_dtype(DT_VARIANT);
209 variant_proto.set_dtype(DT_VARIANT);
252 variant_proto.set_dtype(DT_VARIANT);
variant_test.cc 107 t.set_dtype(DT_FLOAT);
  /external/tensorflow/tensorflow/contrib/gdr/
gdr_worker.cc 102 tensor_proto->set_dtype(val.dtype());
  /external/tensorflow/tensorflow/contrib/session_bundle/
bundle_shim.cc 261 tensor_info.set_dtype(it_tensor_info->second);
291 tensor_info.set_dtype(it_tensor_info->second);
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 92 output_tensor->set_dtype(DT_FLOAT);
113 output_tensor->set_dtype(DT_FLOAT);
229 tensor->set_dtype(DT_INT32);
251 tensor->set_dtype(DT_INT32);
269 tensor->set_dtype(DT_INT32);
287 tensor->set_dtype(DT_INT32);
302 tensor->set_dtype(DT_INT32);
624 max_bounds_const_op_tensor->set_dtype(DT_FLOAT);
633 min_bounds_const_op_tensor->set_dtype(DT_FLOAT);
771 sum_reduction_indices_tensor->set_dtype(DT_INT32)
    [all...]
  /external/tensorflow/tensorflow/cc/ops/
const_op_test.cc 106 proto.set_dtype(DT_DOUBLE);
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_tensor_coding.cc 122 skeleton.set_dtype(val.dtype());
  /external/tensorflow/tensorflow/core/distributed_runtime/
tensor_coding.cc 168 tensor_meta->set_dtype(static_cast<DataType>(static_cast<int>(v)));
  /external/tensorflow/tensorflow/python/framework/
cpp_shape_inference.cc 149 out_shape_and_type->set_dtype(p.dtype);
  /external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/
resolve_svdf.cc 193 allocated_tensor->set_dtype(tensor.dtype());
  /external/tensorflow/tensorflow/contrib/mpi/
mpi_rendezvous_mgr.cc 233 mpi_send_call->mRes_.mutable_response()->mutable_tensor()->set_dtype(
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_rewrite.cc 104 result.set_dtype(DT_INT32);
  /external/tensorflow/tensorflow/core/grappler/optimizers/
auto_parallel.cc 46 tensor->set_dtype(DT_FLOAT);
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_util.cc 144 proto->set_dtype(tensor.dtype());

Completed in 1621 milliseconds

1 2